package com.gan.yidademo.common.util;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Base64;
import java.util.Base64.Decoder;
import java.util.Base64.Encoder;

/**
 * 图片工具类
 */
public class ImageUtil {

    /**
     * BASE64加密网络图片，返回加密之后的字符串
     *
     * @param imageUrl     图片的绝对地址
     * @param extensioName 图片的扩展名，例如 jpg、bmp等
     * @return 加密之后的字符串
     */
    public static String encodeImage2Base64(URL imageUrl, String extensioName) {
        ByteArrayOutputStream outputStream = null;
        try {
            BufferedImage bufferedImage = ImageIO.read(imageUrl);  //读取网络图片
            outputStream = new ByteArrayOutputStream();  //文件输出流
            ImageIO.write(bufferedImage, extensioName, outputStream);
        } catch (MalformedURLException e1) {
            e1.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        Encoder encoder = Base64.getEncoder();

        return encoder.encodeToString(outputStream.toByteArray());// 返回Base64编码过的字节数组字符串  
    }

    /**
     * @Description： 图片转化成base64字符串
     * @param: path
     * @Return:
     */
    public static String GetImageStr(String path) {
        //将图片文件转化为字节数组字符串，并对其进行Base64编码处理
        //待处理的图片

        File imgFile = new File(path);
        InputStream in = null;
        byte[] data = null;
        //读取图片字节数组
        try {
            in = new FileInputStream(imgFile);
            data = new byte[(int) imgFile.length()];
            in.read(data);
            in.close();
        } catch (IOException e) {
            e.printStackTrace();
            return "";
        }
        //对字节数组Base64编码
        Base64.getEncoder().encodeToString(data);
        //返回Base64编码过的字节数组字符串
        return Base64.getEncoder().encodeToString(data);
    }


    public static boolean GenerateImage(String imgStr) {   //对字节数组字符串进行Base64解码并生成图片
        if (imgStr == null) //图像数据为空
            return false;
//        BASE64Decoder decoder = new BASE64Decoder();
        Decoder decoder = Base64.getDecoder();
        try {
            //Base64解码
            byte[] b = decoder.decode(imgStr);
            for (int i = 0; i < b.length; ++i) {
                if (b[i] < 0) {//调整异常数据
                    b[i] += 256;
                }
            }
            //生成jpeg图片
            String imgFilePath = "d://222.jpg";//新生成的图片
            OutputStream out = new FileOutputStream(imgFilePath);
            out.write(b);
            out.flush();
            out.close();
            return true;
        } catch (Exception e) {
            return false;
        }
    }

    /**
     * 将本地图片进行Base64位编码
     *
     * @param imgUrl       图片的url路径，如http://.....xx.jpg
     * @param extensioName 图片的扩展名
     * @return 加密之后的字符串
     */
    public static String encodeImgageToBase64(File imageFile, String extensioName) {// 将图片文件转化为字节数组字符串，并对其进行Base64编码处理  
        ByteArrayOutputStream outputStream = null;
        try {
            BufferedImage bufferedImage = ImageIO.read(imageFile);
            outputStream = new ByteArrayOutputStream();
            ImageIO.write(bufferedImage, extensioName, outputStream);
        } catch (MalformedURLException e1) {
            e1.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        // 对字节数组Base64编码  
        Encoder encoder = Base64.getEncoder();
        return encoder.encodeToString(outputStream.toByteArray());// 返回Base64编码过的字节数组字符串  
    }


    /**
     * 将Base64位编码的图片进行解码，并保存到指定目录
     *
     * @param base64Str 利用base64加密之后的字符串
     * @param path      文件解密之后存放的地址 例如：D：//
     * @param imgName   文件解密之后命名的名称 例如： test.jpg
     */
    public static void decodeBase64ToImage(String base64Str, String path,
                                           String imgName) {
        Decoder decoder = Base64.getDecoder();
        try {
            FileOutputStream write = new FileOutputStream(new File(path
                    + imgName));
            byte[] decoderBytes = decoder.decode(base64Str);
            write.write(decoderBytes);
            write.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}
